import json
import os

from langchain_core.prompts import ChatPromptTemplate

json_result = json.dumps({
    "result": [
        {
            "content": "A",
            "type": "SELECT"
        },
        {
            "content": "AB",
            "type": "SELECT"
        },
        {
            "content": "ACBDE",
            "type": "SELECT"
        },
        {
            "content": "无",
            "type": "FILL"
        }
    ]
})

roles = os.getenv("ROLES","").split(",")

system_template = """
你是一位{role}，你精通你自己专业内容，同时你时刻保持着对新奇事物的好奇心，请你完成如下问卷调查。

问卷标题:{title}
问题:{questions}

说明:
问题中有单选题、多选题、排序题和主观题，回答规范如下:
- 单选题:单选题只能选择一个答案
- 多选题:多选题会有多选的标识，多选题可以选择一个或者多个
- 排序题:排序题会有排序的标识，排序题按照排好的顺序选择全部答案
- 主观题:主观题直接返回主观答案，如果没有则返回"无"

注意:每道题都必须要有答案，且每个问题必须按照规定回答，你不仅需要返回答案还要返回题目类型，类型有"SELECT"、"FILL"

请按照如下json格式返回
{json_result}
"""

prompt_template = ChatPromptTemplate.from_messages([
    ("system", system_template)
])
